协议定义和遵循

1. 什么是协议？

在 Swift 中，协议（Protocol）是一种定义方法、属性和其他要求的蓝图。协议本身不提供实现，而是由遵循协议的类型（如类、结构体或枚举）来实现这些要求。协议使得代码更加模块化和可复用。

1.1 协议的基本语法

protocol MyProtocol {
    // 定义属性和方法
}

2. 定义协议

2.1 定义属性和方法

协议可以定义属性和方法，但不需要提供实现。属性可以是只读的（get）或可读写的（get set）。

protocol Vehicle {
    var numberOfWheels: Int { get }
    func start()
    func stop()
}

2.2 定义初始化器

协议还可以定义初始化器，遵循协议的类型必须实现这些初始化器。

protocol Initializable {
    init(name: String)
}

3. 遵循协议

3.1 类遵循协议

类可以通过在定义时声明遵循某个协议来实现协议中的要求。

class Car: Vehicle {
    var numberOfWheels: Int = 4
    
    func start() {
        print("Car started")
    }
    
    func stop() {
        print("Car stopped")
    }
}

3.2 结构体遵循协议

结构体也可以遵循协议，并实现协议中的要求。

struct Bicycle: Vehicle {
    var numberOfWheels: Int = 2
    
    func start() {
        print("Bicycle started")
    }
    
    func stop() {
        print("Bicycle stopped")
    }
}

3.3 枚举遵循协议

枚举同样可以遵循协议，并实现协议中的要求。

enum Direction: Vehicle {
    case north, south, east, west
    
    var numberOfWheels: Int {
        return 0
    }
    
    func start() {
        print("Direction started")
    }
    
    func stop() {
        print("Direction stopped")
    }
}

4. 协议扩展

4.1 扩展协议

你可以通过扩展协议来为所有遵循该协议的类型提供默认实现。

extension Vehicle {
    func honk() {
        print("Honk!")
    }
}

4.2 条件扩展

你还可以根据类型是否遵循其他协议来提供不同的默认实现。

extension Vehicle where Self: Initializable {
    func start() {
        print("Started with initializable")
    }
}

5. 实践练习

5.1 定义一个协议

定义一个名为 Drawable 的协议，要求实现 draw() 方法。

protocol Drawable {
    func draw()
}

5.2 实现协议

创建一个 Circle 结构体和一个 Rectangle 类，并让它们遵循 Drawable 协议。

struct Circle: Drawable {
    func draw() {
        print("Drawing a circle")
    }
}

class Rectangle: Drawable {
    func draw() {
        print("Drawing a rectangle")
    }
}

5.3 使用协议

创建一个函数，接受一个遵循 Drawable 协议的参数，并调用 draw() 方法。

func drawShape(_ shape: Drawable) {
    shape.draw()
}

let circle = Circle()
let rectangle = Rectangle()

drawShape(circle)      // 输出: Drawing a circle
drawShape(rectangle)   // 输出: Drawing a rectangle

6. 总结

协议是 Swift 中实现代码复用和模块化的重要工具。通过定义协议，你可以确保不同的类型遵循相同的接口，从而提高代码的可维护性和可扩展性。通过扩展协议，你还可以为遵循协议的类型提供默认实现，进一步简化代码。

希望这篇教程能帮助你理解协议的定义和遵循，并在实际编程中应用这些知识。